package _11_整理题目._3_字符串_递归_dfs_bfs_双指针._排列组合._应用_子集和_划分子集;

import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * https://leetcode-cn.com/problems/subsets-ii/
 *
 * 给你一个整数数组 nums ，其中可能包含重复元素，请你返回该数组所有可能的子集（幂集）。
 * 解集 不能 包含重复的子集。返回的解集中，子集可以按 任意顺序 排列。
 *
 *
 */
public class _011_所有子集数_允许输入的元素重复 {

    // 字典序
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        List<Integer> path = new ArrayList<>();
        Arrays.sort(nums);
        dfs(nums, 0, path, result);
        return result;
    }
    public void dfs(int[] nums, int index, List<Integer> path, List<List<Integer>> result) {

        result.add(new ArrayList<>(path));
        if (path.size() == nums.length) return;

        for (int i = index; i < nums.length; i++) {
            if (i > index && nums[i]==nums[i-1]) continue;

            path.add(nums[i]);
            dfs(nums, i+1, path, result);
            path.remove(path.size()-1);
        }
    }

    @Test
    public void main() {
        int[] nums = {1, 2, 3, 4, 5};
        List<List<Integer>> result = subsetsWithDup(nums);
        for (List<Integer> list : result) {
            System.out.println(list);
        }
//        List<List<Integer>> result1 = subsetsWithDup1(nums);
//        for (List<Integer> list : result1) {
//            System.out.println(list);
//        }
    }

    // 个数序
    public List<List<Integer>> subsetsWithDup1(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        List<Integer> path = new ArrayList<>();
        int n = nums.length;
        Arrays.sort(nums);
        for(int k=0; k<=n; k++) {
            dfs1(nums, k, 0, path, result);
        }
        return result;
    }
    public void dfs1(int[] nums, int k, int index, List<Integer> path, List<List<Integer>> result) {
        if(path.size() == k) {
            result.add(new ArrayList<>(path));
            return;
        }

        for(int i=index; i<nums.length; i++) {
            if(i > index && nums[i]==nums[i-1]) continue;

            path.add(nums[i]);
            dfs1(nums, k, i+1, path, result);
            path.remove(path.size()-1);
        }
    }
}
